import numpy as np
from mopo.mask.base import *

OracleStructure = np.array([[0, 0, 1, 1, 1, 1, 0],  # reward
                            [1, 0, 0, 0, 0, 0, 1],  # direction
                            [0, 0, 0, 0, 1, 1, 0],  # v
                            [1, 1, 0, 0, 0, 0, 1],  # vx
                            [1, 1, 0, 0, 0, 0, 1],  # vy
                            [0, 0, 1, 0, 1, 0, 0],  # x
                            [0, 0, 0, 1, 0, 1, 0]])  # y

DenseStructure = np.ones((7, 7))
CausalStructure = np.array([[1, 1, 1, 1, 1, 1, 0],
                            [1, 0, 1, 1, 0, 0, 1],
                            [0, 1, 0, 1, 0, 1, 0],
                            [1, 1, 1, 1, 0, 0, 1],
                            [1, 1, 1, 1, 0, 0, 1],
                            [0, 1, 1, 0, 0, 0, 0],
                            [0, 1, 0, 1, 0, 1, 0]])  # y
